// 导入必要的模块
const jwt = require('jsonwebtoken');
const User = require('../models/User');

// 身份验证中间件
const protect = async (req, res, next) => {
  try {
    let token;
    
    // 从请求头中获取 Bearer token
    if (req.headers.authorization && req.headers.authorization.startsWith('Bearer')) {
      token = req.headers.authorization.split(' ')[1];
    }

    // 如果没有token，返回未授权错误
    if (!token) {
      return res.status(401).json({ message: 'Not authorized, no token' });
    }

    // 验证token并获取用户信息
    const decoded = jwt.verify(token, process.env.JWT_SECRET);
    req.user = await User.findById(decoded.id).select('-password');
    next();
  } catch (error) {
    // token验证失败，返回未授权错误
    res.status(401).json({ message: 'Not authorized, token failed' });
  }
};

// 导出中间件
module.exports = { protect };